"""首页建模"""
from selenium.webdriver.common.by import By

from .add_member_page import AddMemberPage
from .base_page import BasePage


class MainPage(BasePage):
    _base_url = "https://work.weixin.qq.com/wework_admin/frame#index"

    def to_add_member(self):
        """ 访问首页，点击入口【添加成员】，跳转到添加成员页"""
        self.click(By.CSS_SELECTOR,".ww_indexImg_AddMember")
        return AddMemberPage(self.driver)
